package Leetcode100Hot;

import org.junit.Test;

import java.util.HashMap;
import java.util.HashSet;
import java.util.List;
import java.util.Set;

/*
单词拆分
给你一个字符串 s 和一个字符串列表 wordDict 作为字典。如果可以利用字典中出现的一个或多个单词拼接出 s 则返回 true。
注意：不要求字典中出现的单词全部都使用，并且字典中的单词可以重复使用。

示例 1：
输入: s = "leetcode", wordDict = ["leet", "code"]
输出: true
解释: 返回 true 因为 "leetcode" 可以由 "leet" 和 "code" 拼接成。
示例 2：
输入: s = "applepenapple", wordDict = ["apple", "pen"]
输出: true
解释: 返回 true 因为 "applepenapple" 可以由 "apple" "pen" "apple" 拼接成。
     注意，你可以重复使用字典中的单词。
示例 3：
输入: s = "catsandog", wordDict = ["cats", "dog", "sand", "and", "cat"]
输出: false

提示：
1 <= s.length <= 300
1 <= wordDict.length <= 1000
1 <= wordDict[i].length <= 20
s 和 wordDict[i] 仅由小写英文字母组成
wordDict 中的所有字符串 互不相同
 */
public class _21单词拆分 {


    @Test
    public void test() {
        System.out.println(wordBreak("catsandog", List.of("cats", "dog", "sand", "and", "cat")));
    }

    //WA 超时  35/47
    boolean res = false;

    class Trie{
        class TrieNode{
            public boolean isWord = false;
            public HashMap<Character, TrieNode> children = new HashMap<>();
        }
        public TrieNode root;

        public Trie(){
            root = new TrieNode();
        }

        public void insert(String word){
            TrieNode cur = root;
            for(char c : word.toCharArray()){
                if (cur.children.get(c) == null){
                    cur.children.put(c, new TrieNode());
                }
                cur = cur.children.get(c);
            }
            cur.isWord = true;
        }

        public boolean startsWith(String prefix){
            TrieNode cur = root;
            for(char c : prefix.toCharArray()){
                if (cur.children.get(c) == null){
                    return false;
                }
                cur = cur.children.get(c);
            }
            return true;
        }
    }

    public boolean wordBreak(String s, List<String> wordDict) {

        Backtrack(s,0,s.length(),wordDict);
        return res;
    }

    public void Backtrack(String s,int index, int len,List<String> wordDict){
        if (res || index == len){
            return;
        }
        String substring = s.substring(index);
        Trie trie = new Trie();
        trie.insert(substring);

        for (String string : wordDict) {
            if(trie.startsWith(string)){
                index += string.length();
                if (index == len){
                    res = true;
                    break;
                }
                Backtrack(s, index,s.length(),wordDict);
                index -= string.length();
            }
        }
    }

    //官解：DP
    //我们定义 dp[i] 表示字符串 s 前 i 个字符组成的字符串 s[0..i−1] 是否能被空格拆分成若干个字典中出现的单词。
    //dp[i] = dp[j] && check(s[j..i−1])
    //其中 check(s[j..i−1]) 表示子串 s[j..i−1] 是否出现在字典中。
    /*
    作者：力扣官方题解
    链接：https://leetcode.cn/problems/word-break/solutions/302471/dan-ci-chai-fen-by-leetcode-solution/
     */
    public class Solution {
        public boolean wordBreak(String s, List<String> wordDict) {
            Set<String> wordDictSet = new HashSet(wordDict);
            boolean[] dp = new boolean[s.length() + 1];
            dp[0] = true;
            for (int i = 1; i <= s.length(); i++) {
                for (int j = 0; j < i; j++) {
                    if (dp[j] && wordDictSet.contains(s.substring(j, i))) {
                        dp[i] = true;
                        break;
                    }
                }
            }
            return dp[s.length()];
        }
    }

}
